typing-extensions>=4.7.0
cloudpickle
crafter
numpy
jax
jaxlib
flax
dm-haiku
optax
scikit-learn
seaborn
rich
ruamel.yaml
tensorflow-cpu
tensorflow_probability
moviepy
imageio
ipython
ipykernel
hydra-core
wandb
tensorflow-cpu==2.12.0
tensorflow-estimator==2.12.0
tensorflow-io-gcs-filesystem==0.33.0
tensorflow-probability==0.20.1
